package cn.pconline.autoclub.task.util;

import java.util.HashSet;
import java.util.Set;

import net.sourceforge.pinyin4j.PinyinHelper;
import net.sourceforge.pinyin4j.format.HanyuPinyinCaseType;
import net.sourceforge.pinyin4j.format.HanyuPinyinOutputFormat;
import net.sourceforge.pinyin4j.format.HanyuPinyinToneType;
import net.sourceforge.pinyin4j.format.HanyuPinyinVCharType;
import net.sourceforge.pinyin4j.format.exception.BadHanyuPinyinOutputFormatCombination;

/**
 * 中文拼音转换工具类
 * @author zhoutianhuai
 * @ClassName: Pinyin4jUtil
 *
 * 2015年2月10日 下午4:05:38
 */
public class Pinyin4jUtil {

	/**
	 * 获取中文汉字拼音 默认输出
	 * @param chinese
	 * @return
	 */
	public static String getPinyin(String chinese) {
		return getPinyinZh_CN(makeStringByStringSet(chinese));
	}

	/**
	 * 拼音大写输出
	 * @param chinese
	 * @return
	 */
	public static String getPinyinToUpperCase(String chinese) {
		return getPinyinZh_CN(makeStringByStringSet(chinese)).toUpperCase();
	}

	/**
	 * 拼音小写输出
	 * @param chinese
	 * @return
	 */
	public static String getPinyinToLowerCase(String chinese) {
		return getPinyinZh_CN(makeStringByStringSet(chinese)).toLowerCase();
	}

	/**
	 * 首字母大写输出
	 * @param chinese
	 * @return
	 */
	public static String getPinyinFirstToUpperCase(String chinese) {
		return getPinyin(chinese);
	}

	/**
	 * 拼音简拼输出
	 * 
	 * @param chinese
	 * @return
	 */
	public static String getPinyinJianPin(String chinese) {
		return getPinyinConvertJianPin(getPinyin(chinese));
	}

	/**
	 * 字符集转换
	 * @Name: Pinyin4jUtil.java
	 * @param chinese中文汉字
	 * @throws BadHanyuPinyinOutputFormatCombination
	 */
	public static Set<String> makeStringByStringSet(String chinese) {
		char[] chars = chinese.toCharArray();
		if (chinese != null && !chinese.trim().equalsIgnoreCase("")) {
			char[] srcChar = chinese.toCharArray();
			String[][] temp = new String[chinese.length()][];
			for (int i = 0; i < srcChar.length; i++) {
				char c = srcChar[i];

				// 是中文或者a-z或者A-Z转换拼音
				if (String.valueOf(c).matches("[\\u4E00-\\u9FA5]+")) {

					try {
						temp[i] = PinyinHelper.toHanyuPinyinStringArray(
								chars[i], getDefaultOutputFormat());

					} catch (BadHanyuPinyinOutputFormatCombination e) {
						e.printStackTrace();
					}
				} else if (((int) c >= 'A' && (int) c <= 'Z')
						|| ((int) c >= 'a' && (int) c <= 'z')) {
					temp[i] = new String[] { String.valueOf(srcChar[i]) };
				} else {
					temp[i] = new String[] { "" };   
				}
			}
			String[] pingyinArray = Exchange(temp);
			Set<String> zhongWenPinYin = new HashSet<String>();
			for (int i = 0; i < pingyinArray.length; i++) {
				zhongWenPinYin.add(pingyinArray[i]);
			}
			return zhongWenPinYin;
		}
		return null;
	}

	/**
	 * Default Format 默认输出格式
	 * @return
	 */
	public static HanyuPinyinOutputFormat getDefaultOutputFormat() {
		HanyuPinyinOutputFormat format = new HanyuPinyinOutputFormat();
		format.setCaseType(HanyuPinyinCaseType.LOWERCASE);// 小写
		format.setToneType(HanyuPinyinToneType.WITHOUT_TONE);// 没有音调数字
		format.setVCharType(HanyuPinyinVCharType.WITH_U_AND_COLON);// u显示
		return format;
	}

	/**
	 * 
	 * @param strJaggedArray
	 * @return
	 */
	public static String[] Exchange(String[][] strJaggedArray) {
		String[][] temp = DoExchange(strJaggedArray);
		return temp[0];
	}

	/**
	 * 
	 * @param strJaggedArray
	 * @return
	 */
	private static String[][] DoExchange(String[][] strJaggedArray) {
		int len = strJaggedArray.length;
		if (len >= 2) {
			int len1 = strJaggedArray[0].length;
			int len2 = strJaggedArray[1].length;
			int newlen = len1 * len2;
			String[] temp = new String[newlen];
			int Index = 0;
			for (int i = 0; i < len1; i++) {
				for (int j = 0; j < len2; j++) {
					temp[Index] = capitalize(strJaggedArray[0][i])
							+ capitalize(strJaggedArray[1][j]);
					Index++;
				}
			}
			String[][] newArray = new String[len - 1][];
			for (int i = 2; i < len; i++) {
				newArray[i - 1] = strJaggedArray[i];
			}
			newArray[0] = temp;
			return DoExchange(newArray);
		} else {
			return strJaggedArray;
		}
	}

	/**
	 * 首字母大写
	 * @param s
	 * @return
	 */
	public static String capitalize(String s) {     
	       char ch[];     
	       ch = s.toCharArray();  
	     if (ch!= null && ch.length > 0) { //修改[/color]  
	       if (ch[0] >= 'a' && ch[0] <= 'z') {     
	           ch[0] = (char) (ch[0] - 32);     
	       }   
	     }  
	       String newString = new String(ch);     
	       return newString;   
	}
	   
	/**
	 * 字符串集合转换字符串(逗号分隔)
	 * @param stringSet
	 * @return
	 */
	public static String getPinyinZh_CN(Set<String> stringSet) {
		StringBuilder str = new StringBuilder();
		int i = 0;
		for (String s : stringSet) {
			if (i == stringSet.size() - 1) {
				str.append(s);
			} else {
				str.append(s + ",");
			}
			i++;
		}
		return str.toString();
	}

	/**
	 * 获取每个拼音的简称
	 * @param chinese
	 * @return
	 */
	public static String getPinyinConvertJianPin(String chinese) {
		String[] strArray = chinese.split(",");
		StringBuilder sb = new StringBuilder();
		for (String str : strArray) {
			char arr[] = str.toCharArray(); // 将字符串转化成char型数组
			for (int i = 0; i < arr.length; i++) {
				if (arr[i] >= 'A' && arr[i] <= 'Z') { // 判断是否是大写字母
					sb.append(arr[i] + "");
				}
			}
			sb.append(",");
		}
		return sb.toString();
	}
	
	/**
	 * 多音字特殊处理
	 * @param name
	 * @param initial
	 * @return
	 */
	public static String specialHandling(String name, String initial){
		//首字母特殊处理
    	if(name.startsWith("长")){
    		initial = "C";
    	}
    	if(name.contains("阿斯顿")){
    		initial = "A";
    	}
    	if(name.contains("重庆")){
    		initial = "C";
    	}
    	if(name.startsWith("广")){
    		initial = "G";
    	}
    	if(name.startsWith("奇瑞")){
    		initial = "Q";
    	}
    	if(name.contains("红旗")){
    		initial = "H";
    	}
    	if(name.contains("卡威")){
    		initial = "K";
    	}  
    	if(name.contains("奇瑞")){
    		initial = "Q";
    	}
    	if(name.contains("长安")){
    		initial = "C";
    	}
    	return initial;
	}
}